package cn.edu.xjtu.carlWay.dpRepractice.bestTimeToBuyAndSellStockWithCooldown;

/**
 * @author Hydrion-QLz
 * @date 2022-05-03 10:34
 * @description https://leetcode-cn.com/problems/best-time-to-buy-and-sell-stock-with-cooldown/
 * <p>
 * 309. 最佳买卖股票时机含冷冻期
 * 给定一个整数数组prices，其中第  prices[i] 表示第 i 天的股票价格 。
 * <p>
 * 设计一个算法计算出最大利润。在满足以下约束条件下，你可以尽可能地完成更多的交易（多次买卖一支股票）:
 * <p>
 * 卖出股票后，你无法在第二天买入股票 (即冷冻期为 1 天)。
 * 注意：你不能同时参与多笔交易（你必须在再次购买前出售掉之前的股票）。
 */
class Solution {
    public int maxProfit(int[] prices) {
        int n = prices.length;
        // 0:当天持有股票,1:当天不是冷静期并且没有买卖股票,2:当天卖出股票,3:当天是冷静期
        int[][] dp = new int[2][4];// 注意这里一定至少需要存两天的，或者是在for循环中用另外一个数组保存前一天的状态
        dp[0][0] = -prices[0];

        for (int i = 1; i < n; i++) {
            dp[1][0] = Math.max(dp[0][0], Math.max(dp[0][1], dp[0][3]) - prices[i]);
            dp[1][1] = Math.max(dp[0][1], dp[0][3]);
            dp[1][2] = dp[0][0] + prices[i];
            dp[1][3] = dp[0][2];
            for (int j = 0; j < 4; j++) {
                dp[0][j] = dp[1][j];
            }
        }
        return Math.max(dp[0][1], Math.max(dp[0][2], dp[0][3]));
    }
}
